how to create an array with random numbers in python